# CUDA_VISIBLE_DEVICES='0' CUDA_LAUNCH_BLOCKING=1 python -u -m torch.distributed.launch \
#     --nproc_per_node=1 --master_port 11773 \
#     main.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../autodl-tmp/dataset/office31 \
#     --algorithm DAMamba --output ../autodl-tmp/train_output_3 \
#     --dataset Office31 --source-domain dslr --target-domain amazon  \
#     --pretrained ../autodl-tmp/ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 16 --lr 3e-4 --epochs 100 --iter-per-epoch 80 --auto-resume \
#     --loss-weight 0.1  --alpha 0.5

# CUDA_VISIBLE_DEVICES='0' CUDA_LAUNCH_BLOCKING=1 python -u -m torch.distributed.launch \
#     --nproc_per_node=1 --master_port 11773 \
#     main.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../autodl-tmp/dataset/office31 \
#     --algorithm DAMamba --output ../autodl-tmp/train_output_3 \
#     --dataset Office31 --source-domain dslr --target-domain amazon  \
#     --pretrained ../autodl-tmp/ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 16 --lr 3e-4 --epochs 100 --iter-per-epoch 80 --auto-resume \
#     --loss-weight 0.1  --alpha 0.5

# CUDA_VISIBLE_DEVICES='0' CUDA_LAUNCH_BLOCKING=1 python -u -m torch.distributed.launch \
#     --nproc_per_node=1 --master_port 11773 \
#     main.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../autodl-tmp/dataset/office31 \
#     --algorithm DAMamba --output ../autodl-tmp/train_output_with_lmmd \
#     --dataset Office31 --source-domain amazon --target-domain dslr  \
#     --pretrained ../autodl-tmp/ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 16 --lr 3e-4 --epochs 100 --iter-per-epoch 80 --auto-resume \
#     --loss-weight 0.1  --alpha 0.5 --transfer-loss lmmd 

# CUDA_VISIBLE_DEVICES='0' CUDA_LAUNCH_BLOCKING=1 python -u -m torch.distributed.launch \
#     --nproc_per_node=1 --master_port 11773 \
#     main.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../autodl-tmp/dataset/office31 \
#     --algorithm DAMamba --output ../autodl-tmp/train_output_with_lmmd \
#     --dataset Office31 --source-domain amazon --target-domain webcam  \
#     --pretrained ../autodl-tmp/ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 16 --lr 3e-4 --epochs 100 --iter-per-epoch 80 --auto-resume \
#     --loss-weight 0.1  --alpha 0.5 --transfer-loss lmmd 

# CUDA_VISIBLE_DEVICES='0' CUDA_LAUNCH_BLOCKING=1 python -u -m torch.distributed.launch \
#     --nproc_per_node=1 --master_port 11773 \
#     main.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../autodl-tmp/dataset/office31 \
#     --algorithm DAMamba --output ../autodl-tmp/train_output_with_lmmd \
#     --dataset Office31 --source-domain dslr --target-domain webcam  \
#     --pretrained ../autodl-tmp/ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 16 --lr 3e-4 --epochs 100 --iter-per-epoch 80 --auto-resume \
#     --loss-weight 0.1  --alpha 0.5 --transfer-loss lmmd 

    # CUDA_VISIBLE_DEVICES='0' CUDA_LAUNCH_BLOCKING=1 python -u -m torch.distributed.launch \
    # --nproc_per_node=1 --master_port 11773 \
    # main.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
    # --data-path ../autodl-tmp/dataset/office31 \
    # --algorithm DAMamba --output ../autodl-tmp/train_output_with_test \
    # --dataset Office31 --source-domain webcam --target-domain dslr  \
    # --pretrained ../autodl-tmp/ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
    # --batch-size 16 --lr 3e-4 --epochs 100 --iter-per-epoch 80 --auto-resume \
    # --loss-weight 0.1  --alpha 0.5 --transfer-loss lmmd 

# CUDA_VISIBLE_DEVICES='0' CUDA_LAUNCH_BLOCKING=1 python -u -m torch.distributed.launch \
#     --nproc_per_node=1 --master_port 11773 \
#     main_org.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../autodl-tmp/dataset/office31 \
#     --algorithm DAMamba --output ../autodl-tmp/train_output_with_test \
#     --dataset Office31 --source-domain dslr --target-domain amazon  \
#     --pretrained ../autodl-tmp/ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 3e-4 --epochs 100 --iter-per-epoch 80 --auto-resume 

# CUDA_VISIBLE_DEVICES='0' CUDA_LAUNCH_BLOCKING=1 python -u -m torch.distributed.launch \
#     --nproc_per_node=1 --master_port 11773 \
#     main_org.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../autodl-tmp/dataset/office31 \
#     --algorithm DAMamba --output ../autodl-tmp/train_output_lr/test_lr_5 \
#     --dataset Office31 --source-domain dslr --target-domain amazon  \
#     --pretrained ../autodl-tmp/ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 1e-5 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50

# CUDA_VISIBLE_DEVICES='0' CUDA_LAUNCH_BLOCKING=1 python -u -m torch.distributed.launch \
#     --nproc_per_node=1 --master_port 11773 \
#     main_org.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../autodl-tmp/dataset/office31 \
#     --algorithm DAMamba --output ../autodl-tmp/train_output_lr/test_lr_4 \
#     --dataset Office31 --source-domain dslr --target-domain amazon  \
#     --pretrained ../autodl-tmp/ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50

# CUDA_VISIBLE_DEVICES='0' CUDA_LAUNCH_BLOCKING=1 python -u -m torch.distributed.launch \
#     --nproc_per_node=1 --master_port 11773 \
#     main_org.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../autodl-tmp/dataset/office31 \
#     --algorithm DAMamba --output ../autodl-tmp/train_output_lr/test_lr_3 \
#     --dataset Office31 --source-domain dslr --target-domain amazon  \
#     --pretrained ../autodl-tmp/ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 2e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50

# CUDA_VISIBLE_DEVICES='0' CUDA_LAUNCH_BLOCKING=1 python -u -m torch.distributed.launch \
#     --nproc_per_node=1 --master_port 11773 \
#     main_org.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../autodl-tmp/dataset/office31 \
#     --algorithm DAMamba --output ../autodl-tmp/train_output_test_2 \
#     --dataset Office31 --source-domain dslr --target-domain amazon  \
#     --pretrained ../autodl-tmp/ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50 




# CUDA_VISIBLE_DEVICES='0' CUDA_LAUNCH_BLOCKING=1 python -u -m torch.distributed.launch \
#     --nproc_per_node=1 --master_port 11773 \
#     main_org.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_test_1 \
#     --dataset Office31 --source-domain amazon --target-domain dslr \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50 

# CUDA_VISIBLE_DEVICES='1' CUDA_LAUNCH_BLOCKING=1 python -u -m torch.distributed.launch \
#     --nproc_per_node=1 --master_port 11773 \
#     main_org.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_test \
#     --dataset Office31 --source-domain amazon --target-domain webcam  \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50



# CUDA_VISIBLE_DEVICES='0' CUDA_LAUNCH_BLOCKING=1 python -u -m torch.distributed.launch \
#     --nproc_per_node=1 --master_port 11773 \
#     main_org.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_test_1 \
#     --dataset Office31 --source-domain dslr --target-domain amazon \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50 

# CUDA_VISIBLE_DEVICES='1' CUDA_LAUNCH_BLOCKING=1 python -u -m torch.distributed.launch \
#     --nproc_per_node=1 --master_port 11773 \
#     main_org.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_test \
#     --dataset Office31 --source-domain dslr --target-domain webcam  \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50

# CUDA_VISIBLE_DEVICES='1' CUDA_LAUNCH_BLOCKING=1 python -u -m torch.distributed.launch \
#     --nproc_per_node=1 --master_port 11773 \
#     main_org.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_test \
#     --dataset Office31 --source-domain webcam --target-domain amazon  \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50

# CUDA_VISIBLE_DEVICES='1' CUDA_LAUNCH_BLOCKING=1 python -u -m torch.distributed.launch \
#     --nproc_per_node=1 --master_port 11773 \
#     main_org.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_test \
#     --dataset Office31 --source-domain webcam --target-domain dslr  \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50



# CUDA_VISIBLE_DEVICES='1' CUDA_LAUNCH_BLOCKING=1 python -u -m torch.distributed.launch \
#     --nproc_per_node=1 --master_port 11773 \
#     main_org.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_test \
#     --dataset Office31 --source-domain amazon --target-domain dslr \
#     --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50 

# CUDA_VISIBLE_DEVICES='1' CUDA_LAUNCH_BLOCKING=1 python -u -m torch.distributed.launch \
#     --nproc_per_node=1 --master_port 11773 \
#     main_org.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_test \
#     --dataset Office31 --source-domain amazon --target-domain webcam  \
#     --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50

# # CUDA_VISIBLE_DEVICES='0' CUDA_LAUNCH_BLOCKING=1 python -u -m torch.distributed.launch \
# #     --nproc_per_node=1 --master_port 11773 \
# #     main_org.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
# #     --data-path ../dataset/office31 \
# #     --algorithm DAMamba --output ./output/train_output_test \
# #     --dataset Office31 --source-domain dslr --target-domain amazon  \
# #     --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
# #     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50 

# CUDA_VISIBLE_DEVICES='1' CUDA_LAUNCH_BLOCKING=1 python -u -m torch.distributed.launch \
#     --nproc_per_node=1 --master_port 11773 \
#     main_org.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_test \
#     --dataset Office31 --source-domain dslr --target-domain webcam  \
#     --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50

# CUDA_VISIBLE_DEVICES='1' CUDA_LAUNCH_BLOCKING=1 python -u -m torch.distributed.launch \
#     --nproc_per_node=1 --master_port 11773 \
#     main_org.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_test \
#     --dataset Office31 --source-domain webcam --target-domain amazon  \
#     --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50

# CUDA_VISIBLE_DEVICES='1' CUDA_LAUNCH_BLOCKING=1 python -u -m torch.distributed.launch \
#     --nproc_per_node=1 --master_port 11773 \
#     main_org.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_test \
#     --dataset Office31 --source-domain webcam --target-domain dslr  \
#     --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50


# # augmix
# CUDA_VISIBLE_DEVICES=1 CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11775 \
#     main_org.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_with_weight_1v1 \
#     --dataset Office31 --source-domain amazon --target-domain dslr \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50 \
#     --loss-weight 10.0

# CUDA_VISIBLE_DEVICES=1 CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11775 \
#     main_org.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_with_weight_1v1 \
#     --dataset Office31 --source-domain amazon --target-domain dslr \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50 \
#     --loss-weight 0.7

# CUDA_VISIBLE_DEVICES=1 CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11775 \
#     main_org.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_with_weight_1v1 \
#     --dataset Office31 --source-domain amazon --target-domain dslr \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50 \
#     --loss-weight 10.0

# CUDA_VISIBLE_DEVICES=1 CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11775 \
#     main_org.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_with_weight_1 \
#     --dataset Office31 --source-domain dslr --target-domain amazon \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50 

# CUDA_VISIBLE_DEVICES=1 CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11775 \
#     main_org.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_with_weight_3 \
#     --dataset Office31 --source-domain dslr --target-domain amazon \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50 \
#     --loss-weight 0.3


# CUDA_VISIBLE_DEVICES=1 CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11775 \
#     main_org.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_with_weight_5 \
#     --dataset Office31 --source-domain dslr --target-domain amazon \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50 \
#     --loss-weight 0.5

# CUDA_VISIBLE_DEVICES=1 CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11775 \
#     main_org.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_with_weight_7 \
#     --dataset Office31 --source-domain dslr --target-domain amazon \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50 \
#     --loss-weight 0.7

# CUDA_VISIBLE_DEVICES=1 CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11775 \
#     main_org.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_with_weight_9 \
#     --dataset Office31 --source-domain dslr --target-domain amazon \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50 \
#     --loss-weight 0.9

# CUDA_VISIBLE_DEVICES=1 CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11775 \
#     main_org.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_with_weight_20 \
#     --dataset Office31 --source-domain dslr --target-domain amazon \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50 \
#     --loss-weight 2.0

# CUDA_VISIBLE_DEVICES=1 CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11775 \
#     main_org.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_with_weight_100 \
#     --dataset Office31 --source-domain dslr --target-domain amazon \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50 \
#     --loss-weight 10.0


CUDA_VISIBLE_DEVICES=1 CUDA_LAUNCH_BLOCKING=1 torchrun \
    --nproc-per-node=1 --master-port=11775 \
    uda_main.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
    --data-path ../dataset/office31 \
    --algorithm DAMamba --output ./output/train_output_with_weight_150 \
    --dataset Office31 --source-domain dslr --target-domain amazon \
    --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
    --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50 \
    --loss-weight 15.0

# CUDA_VISIBLE_DEVICES=1 CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11775 \
#     uda_main.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_with_weight_200 \
#     --dataset Office31 --source-domain dslr --target-domain amazon \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50 \
#     --loss-weight 20.0

# CUDA_VISIBLE_DEVICES=1 CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11775 \
#     uda_main.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_with_weight_250v1 \
#     --dataset Office31 --source-domain dslr --target-domain amazon \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 60 --iter-per-epoch 80 --auto-resume --save-freq 30 \
#     --loss-weight 25.0

# CUDA_VISIBLE_DEVICES=1 CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11775 \
#     uda_main.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_with_weight_150 \
#     --dataset Office31 --source-domain amazon --target-domain dslr \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50 \
#     --loss-weight 15.0

# CUDA_VISIBLE_DEVICES=1 CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11775 \
#     uda_main.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_with_weight_200 \
#     --dataset Office31 --source-domain amazon --target-domain dslr \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50 \
#     --loss-weight 20.0

# CUDA_VISIBLE_DEVICES=1 CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11775 \
#     main.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm MixStyle --output ./output/train_output_with_layermemoryv3 \
#     --dataset Office31 --source-domain amazon --target-domain dslr \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50

# CUDA_VISIBLE_DEVICES=1 CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11775 \
#     main.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm MixStyle --output ./output/train_output_with_layermemoryv3 \
#     --dataset Office31 --source-domain amazon --target-domain webcam \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50

# CUDA_VISIBLE_DEVICES=1 CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11775 \
#     main.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm MixStyle --output ./output/train_output_with_layermemoryv3 \
#     --dataset Office31 --source-domain dslr --target-domain amazon \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50


# CUDA_VISIBLE_DEVICES=1 CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11775 \
#     main.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm MixStyle --output ./output/train_output_with_layermemoryv3 \
#     --dataset Office31 --source-domain dslr --target-domain webcam \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50

# CUDA_VISIBLE_DEVICES=1 CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11775 \
#     main.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm MixStyle --output ./output/train_output_with_layermemoryv3 \
#     --dataset Office31 --source-domain webcam --target-domain amazon \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50

# CUDA_VISIBLE_DEVICES=1 CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11775 \
#     main.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm MixStyle --output ./output/train_output_with_layermemoryv3 \
#     --dataset Office31 --source-domain webcam --target-domain dslr \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50

# CUDA_VISIBLE_DEVICES=1 CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11775 \
#     main.py --cfg ./configs/vssm/vmambav2_base_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm MixStyle --output ./output/train_output_with_layermemoryv1 \
#     --dataset Office31 --source-domain webcam --target-domain amazon \
#     --pretrained ./ckpts/vssm_base_0229_ckpt_epoch_237.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50


# CUDA_VISIBLE_DEVICES=2 CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11775 \
#     main.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm MixStyle --output ./output/train_output_with_layermemoryv7 \
#     --dataset Office31 --source-domain amazon --target-domain dslr \
#     --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50


# CUDA_VISIBLE_DEVICES=2 CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11775 \
#     main.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm MixStyle --output ./output/train_output_with_layermemoryv7 \
#     --dataset Office31 --source-domain amazon --target-domain webcam \
#     --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50



# CUDA_VISIBLE_DEVICES=2 CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11775 \
#     main.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm MixStyle --output ./output/train_output_with_layermemoryv7 \
#     --dataset Office31 --source-domain dslr --target-domain amazon \
#     --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50


# CUDA_VISIBLE_DEVICES=2 CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11775 \
#     main.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm MixStyle --output ./output/train_output_with_layermemoryv7 \
#     --dataset Office31 --source-domain dslr --target-domain webcam \
#     --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50



# CUDA_VISIBLE_DEVICES=2 CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11775 \
#     main.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm MixStyle --output ./output/train_output_with_layermemoryv7 \
#     --dataset Office31 --source-domain webcam --target-domain amazon \
#     --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50


# CUDA_VISIBLE_DEVICES=2 CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11775 \
#     main.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm MixStyle --output ./output/train_output_with_layermemoryv7 \
#     --dataset Office31 --source-domain webcam --target-domain dslr \
#     --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50


# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc_per_node=1 --master_port 11777 \
#     uda_main.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_with_patchmmd \
#     --dataset Office31 --source-domain amazon --target-domain dslr  \
#     --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50 \
#     --loss-weight 0.5

# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc_per_node=1 --master_port 11777 \
#     uda_main.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_with_patchcoral1 \
#     --dataset Office31 --source-domain amazon --target-domain dslr  \
#     --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50 \
#     --loss-weight 0.1

# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc_per_node=1 --master_port 11777 \
#     uda_main.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_with_CMPLoss \
#     --dataset Office31 --source-domain amazon --target-domain dslr  \
#     --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50 \
#     --loss-weight 1.0

# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc_per_node=1 --master_port 11777 \
#     uda_main.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_with_CMPLoss \
#     --dataset Office31 --source-domain dslr --target-domain amazon  \
#     --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50 \
#     --loss-weight 1.0

# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc_per_node=1 --master_port 11777 \
#     uda_main.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_with_CMPLoss1 \
#     --dataset Office31 --source-domain amazon --target-domain dslr  \
#     --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50 \
#     --loss-weight 0.3

# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc_per_node=1 --master_port 11777 \
#     uda_main.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../dataset/office31 \
#     --algorithm DAMamba --output ./output/train_output_with_CMPLoss1 \
#     --dataset Office31 --source-domain dslr --target-domain amazon  \
#     --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 80 --auto-resume --save-freq 50 \
#     --loss-weight 0.3

# tiny office home
# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11777 \
#     uda_main.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../dataset/OfficeHome \
#     --algorithm DAMamba --output ./output/train_output_with_CMPLoss1 \
#     --dataset OfficeHome --source-domain Art --target-domain Clipart \
#     --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 3e-4 --epochs 100 --iter-per-epoch 100 --auto-resume --save-freq 50 

# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11777 \
#     uda_main.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../dataset/OfficeHome \
#     --algorithm DAMamba --output ./output/train_output_with_CMPLoss1  \
#     --dataset OfficeHome --source-domain Art --target-domain Product \
#     --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 3e-4 --epochs 100 --iter-per-epoch 100 --auto-resume --save-freq 50 

# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11777 \
#     uda_main.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../dataset/OfficeHome \
#     --algorithm DAMamba --output ./output/train_output_with_CMPLoss1 \
#     --dataset OfficeHome --source-domain Art --target-domain RealWorld \
#     --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 3e-4 --epochs 100 --iter-per-epoch 100 --auto-resume --save-freq 50 

# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11777 \
#     uda_main.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../dataset/OfficeHome \
#     --algorithm DAMamba --output ./output/train_output_with_CMPLoss1  \
#     --dataset OfficeHome --source-domain Clipart --target-domain Art \
#     --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 3e-4 --epochs 100 --iter-per-epoch 100 --auto-resume --save-freq 50 

# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11777 \
#     uda_main.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../dataset/OfficeHome \
#     --algorithm DAMamba --output ./output/train_output_with_CMPLoss1 \
#     --dataset OfficeHome --source-domain Clipart --target-domain Product \
#     --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 3e-4 --epochs 100 --iter-per-epoch 100 --auto-resume --save-freq 50 

# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11777 \
#     uda_main.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../dataset/OfficeHome \
#     --algorithm DAMamba --output ./output/train_output_with_CMPLoss1  \
#     --dataset OfficeHome --source-domain Clipart --target-domain RealWorld \
#     --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 3e-4 --epochs 100 --iter-per-epoch 100 --auto-resume --save-freq 50 

# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11777 \
#     uda_main.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../dataset/OfficeHome \
#     --algorithm DAMamba --output ./output/train_output_with_CMPLoss1 \
#     --dataset OfficeHome --source-domain Product --target-domain Art \
#     --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 3e-4 --epochs 100 --iter-per-epoch 100 --auto-resume --save-freq 50 

# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11777 \
#     uda_main.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../dataset/OfficeHome \
#     --algorithm DAMamba --output ./output/train_output_with_CMPLoss1  \
#     --dataset OfficeHome --source-domain Product --target-domain Clipart \
#     --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 3e-4 --epochs 100 --iter-per-epoch 100 --auto-resume --save-freq 50 

# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11777 \
#     uda_main.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../dataset/OfficeHome \
#     --algorithm DAMamba --output ./output/train_output_with_CMPLoss1 \
#     --dataset OfficeHome --source-domain Product --target-domain RealWorld \
#     --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 3e-4 --epochs 100 --iter-per-epoch 100 --auto-resume --save-freq 50 

# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11777 \
#     uda_main.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../dataset/OfficeHome \
#     --algorithm DAMamba --output ./output/train_output_with_CMPLoss1  \
#     --dataset OfficeHome --source-domain RealWorld --target-domain Art \
#     --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 3e-4 --epochs 100 --iter-per-epoch 100 --auto-resume --save-freq 50 

# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11777 \
#     uda_main.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../dataset/OfficeHome \
#     --algorithm DAMamba --output ./output/train_output_with_CMPLoss1 \
#     --dataset OfficeHome --source-domain RealWorld --target-domain Clipart \
#     --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 3e-4 --epochs 100 --iter-per-epoch 100 --auto-resume --save-freq 50 

# CUDA_VISIBLE_DEVICES='2' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11777 \
#     uda_main.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../dataset/OfficeHome \
#     --algorithm DAMamba --output ./output/train_output_with_CMPLoss1  \
#     --dataset OfficeHome --source-domain RealWorld --target-domain Product \
#     --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 3e-4 --epochs 100 --iter-per-epoch 100 --auto-resume --save-freq 50 

# CUDA_VISIBLE_DEVICES='1' CUDA_LAUNCH_BLOCKING=1 torchrun \
#     --nproc-per-node=1 --master-port=11778 \
#     uda_main.py --cfg ./configs/vssm/vmambav2_tiny_224.yaml \
#     --data-path ../dataset/visda17 \
#     --algorithm DAMamba --output ./output/train_output_with_CMPLoss3  \
#     --dataset Visda17 --source-domain train --target-domain validation \
#     --pretrained ./ckpts/vssm_tiny_0230_ckpt_epoch_262.pth \
#     --batch-size 32 --lr 1e-4 --epochs 100 --iter-per-epoch 200 --auto-resume --save-freq 50 